package hutech.ksv.entity;

import java.util.List;

import javax.faces.model.ListDataModel;

import org.primefaces.model.SelectableDataModel;

public class CongTyDataModel extends ListDataModel<ThongTinCongTyEntity> implements
		SelectableDataModel<ThongTinCongTyEntity> {
	
	public CongTyDataModel() {
		
	}
	
	public CongTyDataModel(List<ThongTinCongTyEntity> data) {
		super(data);
	}

	@Override
	public ThongTinCongTyEntity getRowData(String rowKey) {
		List<ThongTinCongTyEntity> danhSachCongTy = (List<ThongTinCongTyEntity>) getWrappedData();
		
		for(ThongTinCongTyEntity c : danhSachCongTy) {
			if(c.getMaCTTD() == Integer.parseInt(rowKey)) {
				return c;
			}
		}
		return null;
	}

	@Override
	public Object getRowKey(ThongTinCongTyEntity thongTinCongTyEntity) {
		return thongTinCongTyEntity.getMaCTTD();
	}

}
